package com.example.csetpp.utils;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;

import org.springframework.mock.web.MockMultipartFile;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 文件解析和存储工具类
 */
public class FileParserUtil {

    /**
     * 解析JSON或CSV文件，返回JSON对象
     * @param filePath 文件路径
     * @return 解析后的JSON对象
     * @throws IOException 文件读取异常
     */
    public static Object parseFile(String filePath) throws IOException {
        if (StringUtils.isEmpty(filePath)) {
            throw new IllegalArgumentException("文件路径不能为空");
        }

        String fileExtension = filePath.substring(filePath.lastIndexOf(".")).toLowerCase();
        
        switch (fileExtension) {
            case ".json":
                return parseJsonFile(filePath);
            case ".csv":
                return parseCsvFile(filePath);
            default:
                throw new IllegalArgumentException("不支持的文件格式: " + fileExtension);
        }
    }

    /**
     * 解析JSON或CSV文件，返回JSON对象
     * @param file 上传的文件
     * @return 解析后的JSON对象
     * @throws IOException 文件读取异常
     */
    public static Object parseFile(MultipartFile file) throws IOException {
        if (file == null || file.isEmpty()) {
            throw new IllegalArgumentException("文件不能为空");
        }

        String originalFilename = file.getOriginalFilename();
        if (originalFilename == null || originalFilename.isEmpty()) {
            throw new IllegalArgumentException("文件名不能为空");
        }

        String fileExtension = originalFilename.substring(originalFilename.lastIndexOf(".")).toLowerCase();
        
        switch (fileExtension) {
            case ".json":
                return parseJsonFile(file);
            case ".csv":
                return parseCsvFile(file);
            default:
                throw new IllegalArgumentException("不支持的文件格式: " + fileExtension);
        }
    }

    /**
     * 解析JSON文件
     * @param file JSON文件
     * @return 解析后的JSON对象
     * @throws IOException 文件读取异常
     */
    private static Object parseJsonFile(MultipartFile file) throws IOException {
        String content = new String(file.getBytes(), StandardCharsets.UTF_8);
        return JSON.parse(content);
    }

    /**
     * 解析CSV文件
     * @param file CSV文件
     * @return 解析后的JSON数组
     * @throws IOException 文件读取异常
     */
    private static JSONArray parseCsvFile(MultipartFile file) throws IOException {
        JSONArray jsonArray = new JSONArray();
        
        try (Reader reader = new InputStreamReader(file.getInputStream(), StandardCharsets.UTF_8);
             CSVParser csvParser = new CSVParser(reader, CSVFormat.DEFAULT.withFirstRecordAsHeader())) {
            
            // 获取CSV文件的表头
            Map<String, Integer> headerMap = csvParser.getHeaderMap();
            
            // 遍历每一行记录
            for (CSVRecord record : csvParser) {
                JSONObject jsonObject = new JSONObject();
                
                // 将每一行的数据添加到JSON对象中
                for (Map.Entry<String, Integer> entry : headerMap.entrySet()) {
                    String header = entry.getKey();
                    int index = entry.getValue();
                    jsonObject.put(header, record.get(index));
                }
                
                jsonArray.add(jsonObject);
            }
        }
        
        return jsonArray;
    }

    /**
     * 将JSON数据转换为MultipartFile
     * @param jsonData JSON数据对象
     * @param fileName 文件名称（不包含路径）
     * @return MultipartFile类型的文件
     * @throws IOException 文件处理异常
     */
    public static MultipartFile convertJsonToMultipartFile(Object jsonData, String fileName) throws IOException {
        if (jsonData == null) {
            throw new IllegalArgumentException("JSON数据不能为空");
        }
        
        if (StringUtils.isEmpty(fileName)) {
            fileName = "data.json";
        } else if (!fileName.toLowerCase().endsWith(".json")) {
            fileName += ".json";
        }
        
        // 将JSON对象转换为格式化的JSON字符串
        String jsonString;
        if (jsonData instanceof String) {
            // 如果已经是字符串，尝试解析并格式化
            try {
                Object parsedJson = JSON.parse((String) jsonData);
                jsonString = JSON.toJSONString(parsedJson, true);
            } catch (Exception e) {
                // 如果解析失败，直接使用原字符串
                jsonString = (String) jsonData;
            }
        } else {
            // 否则直接格式化对象
            jsonString = JSON.toJSONString(jsonData, true);
        }
        
        // 将字符串转换为字节数组
        byte[] content = jsonString.getBytes(StandardCharsets.UTF_8);
        
        // 创建MultipartFile
        return new MockMultipartFile(
                fileName,                      // 文件名
                fileName,                      // 原始文件名
                "application/json",            // 内容类型
                content                        // 文件内容
        );
    }

    /**
     * 解析JSON文件
     * @param filePath JSON文件路径
     * @return 解析后的JSON对象
     * @throws IOException 文件读取异常
     */
    private static Object parseJsonFile(String filePath) throws IOException {
        String content = new String(Files.readAllBytes(Paths.get(filePath)), StandardCharsets.UTF_8);
        return JSON.parse(content);
    }

    /**
     * 解析CSV文件
     * @param filePath CSV文件路径
     * @return 解析后的JSON数组
     * @throws IOException 文件读取异常
     */
    private static JSONArray parseCsvFile(String filePath) throws IOException {
        JSONArray jsonArray = new JSONArray();
        
        try (Reader reader = Files.newBufferedReader(Paths.get(filePath));
             CSVParser csvParser = new CSVParser(reader, CSVFormat.DEFAULT.withFirstRecordAsHeader())) {
            
            // 获取CSV文件的表头
            Map<String, Integer> headerMap = csvParser.getHeaderMap();
            
            // 遍历每一行记录
            for (CSVRecord record : csvParser) {
                JSONObject jsonObject = new JSONObject();
                
                // 将每一行的数据添加到JSON对象中
                for (Map.Entry<String, Integer> entry : headerMap.entrySet()) {
                    String header = entry.getKey();
                    int index = entry.getValue();
                    jsonObject.put(header, record.get(index));
                }
                
                jsonArray.add(jsonObject);
            }
        }
        
        return jsonArray;
    }

    /**
     * 将JSON数据转换为MultipartFile（不保存到本地）
     * @param jsonData JSON数据对象
     * @param fileName 文件名称（不包含路径）
     * @return MultipartFile类型的文件
     * @throws IOException 文件处理异常
     */
    public static MultipartFile saveJsonToFile(Object jsonData, String fileName) throws IOException {
        return convertJsonToMultipartFile(jsonData, fileName);
    }
}